문제 설명

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.

예를 들어, 문자열 S = baabaa 라면

b aa baa → bb aa → aa

의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

제한사항

  • 문자열의 길이 : 1,000,000이하의 자연수
  • 문자열은 모두 소문자로 이루어져 있습니다.

입출력 예

s result
baabaa 1
cdcd 0
입출력 예 설명

입출력 예 #1 위의 예시와 같습니다. 입출력 예 #2 문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.

나의 풀이

첫번째 풀이

소스

from collections import deque


def solution(s):
    stack = deque([])

    for c in s:
        stack.append(c)
        if len(stack) >= 2 and stack[-1] == stack[-2]:
            stack.pop()
            stack.pop()

    if stack:
        return 0
    else:
        return 1

설명

  • Stack을 이용하여서 쉽게 풀 수 있었다.
  • stack을 이용해서 끝의 2개를 비교한 후 같으면 둘 다 pop합니다.
  • 마지막에 stack이 비어있으면 1을 리턴합니다.
  • list pop시에 O(n)이 소요된다고 하셔서 알려주신 deque를 이용하였습니다.
  • 질문
    • 문득 궁금해졌는데, 왜 list pop(0)이나 pop()이나 O(n)인가요? 어차피 시작과 끝에있는걸 제거하는 것이니 O(1)이 걸려야 되는게 아닌가요? 왜 O(n)으로 비효율적으로 제거를 하는걸까요?

결과

통과

리뷰

  • return 0 if stack else 1로 줄일 수 있습니다. :)
  • 조건을 조금 바꿔서 stack.pop()을 두 번하지 않도록 수정할 수 있습니다. :)

    for c in s:
      if stack and stack[-1] == c:
          stack.pop()
      else:
          stack.append(c)
  • deque를 사용하지 않아도 괜찮습니다. :)
  • 답변

    list의 pop(0)은 O(n)이지만 pop()은 O(1)입니다. :) 이렇게 구현된 이유는 python의 list가 성능을 위해 C언어의 pointer array로 구현되서 그렇습니다. pop(0)을 했을 때 값을 리턴하는 것은 O(1)이지만 각 요소들을 앞으로 한 칸씩 당겨줘야하기 때문입니다. 예를들어 [5, 1, 6, 2, 9, 8]이라는 값이 있을 때 1을 제거하면 [5, X, 6, 2, 9, 8] 상태가되고 각 6, 2, 9, 8을 한 칸씩 당겨줘야 합니다. 그렇기 때문에 0번째 값을 제거한다면 결과적으로 O(n)의 시간복잡도가 발생합니다. :)

스터디 리더의 풀이

def solution(s):
    stack = []

    for c in s:
        if stack and stack[-1] == c:
            stack.pop()
        else:
            stack.append(c)

    return int(stack == [])